package com.ebupt.migu.music.label.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.text.DecimalFormat;

/**
 * 文件大小获取工具类
 *
 * @author lemon
 * @since 2022-08-29 16:57
 */
public class FileSizeUtil {

    /**
     * 获取文件大小单位为B的double值
     */
    private static final int SIZE_TYPE_B = 1;

    /**
     * 获取文件大小单位为KB的double值
     */
    private static final int SIZE_TYPE_KB = 2;

    /**
     * 获取文件大小单位为MB的double值
     */
    private static final int SIZE_TYPE_MB = 3;

    /**
     * 获取文件大小单位为GB的double值
     */
    private static final int SIZE_TYPE_GB = 4;

    /**
     * 获取文件指定文件的指定单位的大小
     *
     * @param filePath 文件路径
     * @param sizeType 获取大小的类型1为B、2为KB、3为MB、4为GB
     * @return double值的大小
     */
    public static double getFileOrFilesSize(String filePath, int sizeType) {
        File file = new File(filePath);
        long blockSize = 0;
        try {
            if (file.isDirectory()) {
                blockSize = getFileSizes(file);
            } else {
                blockSize = getFileSize(file);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return formatFileSize(blockSize, sizeType);
    }

    /**
     * 转换文件大小,指定转换的类型
     *
     * @param files 文件大小
     * @param sizeType 转换类型
     * @return 转换后文件大小
     */
    private static double formatFileSize(long files, int sizeType) {
        DecimalFormat decimalFormat = new DecimalFormat("#.00");
        double fileSizeLong = 0;
        switch (sizeType) {
            case SIZE_TYPE_B:
                fileSizeLong = Double.parseDouble(decimalFormat.format((double) files));
                break;
            case SIZE_TYPE_KB:
                fileSizeLong = Double.parseDouble(decimalFormat.format((double) files / 1024));
                break;
            case SIZE_TYPE_MB:
                fileSizeLong = Double.parseDouble(decimalFormat.format((double) files / 1048576));
                break;
            case SIZE_TYPE_GB:
                fileSizeLong = Double.parseDouble(decimalFormat.format((double) files / 1073741824));
                break;
            default:
                break;
        }
        return fileSizeLong;
    }

    /**
     * 获取指定文件大小
     *
     * @param file 文件
     * @return 文件大小
     * @throws IOException 异常处理
     */
    private static long getFileSize(File file) throws IOException {
        long size = 0;
        if (file.exists()) {
            FileInputStream fis = new FileInputStream(file);
            size = fis.available();
            fis.close();
        } else {
            file.createNewFile();
        }
        return size;
    }

    /**
     * 获取指定文件夹
     *
     * @param file 文件夹
     * @return 文件夹大小
     */
    private static long getFileSizes(File file) throws IOException {
        long size = 0;
        File[] files = file.listFiles();
        assert files != null;
        for (File f : files) {
            if (f.isDirectory()) {
                size = size + getFileSizes(f);
            } else {
                size = size + getFileSize(f);
            }
        }
        return size;
    }
}